//
//  UIButton+ACCountDown.m
//  ACFramework
//
//  Created by caoyq on 2018/11/22.
//

#import "UIButton+ACCountDown.h"

@implementation UIButton (ACCountDown)

- (void)ac_addCountDownWithDuration: (NSInteger)duration waitTitle: (NSString *)waitTitle completion:(void (^)(void))completion
{
    NSString *originTitle = self.titleLabel.text;
    __block NSInteger timeOut = duration; //倒计时时间
    dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
    dispatch_source_t _timer = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0,queue);
    dispatch_source_set_timer(_timer,dispatch_walltime(NULL, 0),1.0*NSEC_PER_SEC, 0); //每秒执行
    dispatch_source_set_event_handler(_timer, ^{
        
        if(timeOut <= 0)
        { //倒计时结束，关闭
            dispatch_source_cancel(_timer);
            dispatch_async(dispatch_get_main_queue(), ^{
                //按钮文字还原
                [self setTitle:originTitle forState:UIControlStateNormal];
                self.userInteractionEnabled = YES;
    
                if (completion) {
                    completion();
                }
            });
        }
        else
        {
            NSString *timeStr = [NSString stringWithFormat:@"%ld", timeOut];
            dispatch_async(dispatch_get_main_queue(), ^{
                //按钮显示倒计时组合文字
                [self setTitle:[NSString stringWithFormat:@"%@%@",timeStr,waitTitle] forState:UIControlStateNormal];
                self.userInteractionEnabled = NO;
            });
            timeOut--;
        }
    });
    dispatch_resume(_timer);
}

@end
